#pragma once

#include <string>
#include <vector>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/time.h>
#include <atomic>
#include <fstream>
#include <boost/algorithm/string.hpp>


namespace ns_util
{

    class TimeUtil
    {
    public:
        // 获取时间戳
        static std::string GetTimeStamp()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec);
        }

        // 通过毫秒级时间戳+原子性递增生成唯一文件名
        static std::string GetTimeMs()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec * 1000 + _time.tv_usec / 1000);
        }
    };

    const std::string temp_path = "./temp/"; // 存储编译运行临时文件的目录

    class PathUtil
    {
    public:
        // 进行实际的文件路径和文件名和后缀的拼接
        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string path_name = temp_path;
            path_name += file_name;
            path_name += suffix;
            return path_name;
        }

        //编译时需要的临时文件
        // 构建源文件路径+后缀的完整文件名
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }
        static std::string CompilerError(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compiler_err");
        } 
        static std::string Exe(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }

        //运行时需要的临时文件
        static std::string Stdin(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }
        static std::string Stdout(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }
        static std::string Stderr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
    };

    class FileUtil
    {
    public:
        // 判断文件是否存在
        static bool IsFileExists(const std::string &path_name)
        {
            struct stat st; // 备用，用于存储文件的属性信息
            // stat检测特定路径下对应的文件，获取它的属性
            if(stat(path_name.c_str(), &st) == 0)
            {
                // 获取属性成功，说明文件已经存在
                return true; 
            }
            return false;
        }

        // 通过毫秒级时间戳+原子性递增唯一值，来保证唯一性
        static std::string UniqueFile()
        {
            static std::atomic_uint id(0);
            id++;
            std::string ms = TimeUtil::GetTimeMs();
            std::string uniq_id = std::to_string(id);
            return ms + "_" + uniq_id;
        }

        // 向文件中写入
        static bool WriteFile(const std::string &path_name, const std::string &content)
        {
            std::ofstream out(path_name);
            if(!out.is_open())
            {
                return false;
            }

            out.write(content.c_str(), content.size());
            out.close();
            return true;
        }

        // 从文件file_name中读取
        // content输出型参数，获取读取的数据，keep表示是否保留行分隔符
        static bool ReadFile(const std::string &path_name, std::string *content, bool keep = false)
        {
            std::ifstream in(path_name);
            if(!in.is_open())
            {
                return false;
            }

            std::string line;
            // getline不保存行分隔符，但有些情况我们想要保留\n
            // getline内部重载了强制类型转换
            while(getline(in, line))
            {
                (*content) += line;
                (*content) += (keep ? "\n" : "");
            }

            in.close();
            return true;
        }
    };

    class StringUtil
    {
    public:
        /***********************************
         * str: 要分割的字符串
         * tar: 输出型参数，带出分割后的多个字符串
         * sep: 分隔符
        ************************************/
        static void SplitString(const std::string &str, std::vector<std::string> *tar, const std::string & sep)
        {
            /***************************************
             * tar存储切割后的子串
             * str表示要进行切割的字符串
             * boost::is_any_of(sep)表示遇到sep里的分隔符就进行切割
             * 最后一个参数表示不保留两个相邻分隔符之间的空内容
            ***************************************/
            boost::split(*tar, str, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };


}